package com.adminserver.config;

import cn.dev33.satoken.context.SaHolder;
import cn.dev33.satoken.exception.NotLoginException;
import cn.dev33.satoken.filter.SaServletFilter;
import cn.dev33.satoken.router.SaRouter;
import cn.dev33.satoken.stp.StpUtil;
import com.adminserver.util.HandlerException.HandlerException;
import com.adminserver.util.responseUtil.ResponseEnum;
import com.adminserver.util.responseUtil.ResponseUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.ArrayList;

@Configuration
public class SaTokenConfigure {

    @Autowired
    HandlerException handlerException;

    static final ArrayList<String> excludePattern = new ArrayList<>();

    static {
        excludePattern.add("/login/getKey");
        excludePattern.add("/login/start");
        excludePattern.add("/realTimeService/**");
    }

    /**
     * 全局过滤器
     *
     * @return
     */
    @Bean
    public SaServletFilter saServletFilter() {
        SaServletFilter saServletFilter = new SaServletFilter();

        // 指定 拦截路由 与 放行路由
        saServletFilter.addInclude("/**");
        saServletFilter.addExclude("/favicon.ico");

        // 认证函数: 每次请求执行
        saServletFilter.setAuth(authRun -> {
            String satoken = SaHolder.getRequest().getHeader("satoken");
            System.out.println("---------- 进入Sa-Token全局认证 -----------");
            System.out.println("satoken: " + satoken);

            //登录认证 -- 拦截所有路由，并排除 指定地址 用于开放登录
            SaRouter.match("/**").notMatch(excludePattern)
                    .check(checkRun -> {
                        try {
                            StpUtil.checkLogin();
                        } catch (NotLoginException nle) {
                            checkRun.back(handlerException.handlerNotLogin(nle));
                        }
                    });
            // -------------------------------角色验证-------------------------------
            /**
             * 角色验证--访问系统除特定接口的所有接口皆需要为管理员
             */
            SaRouter.match("/**").notMatch(excludePattern).check(check -> {
                try {
                    StpUtil.checkRole("管理员");
                } catch (Exception e) {
                    check.back(handlerException.handlerRolePermission(e));
                }
            });

            // -------------------------------权限验证-------------------------------

            /**
             * 首页 菜单权限
             */
            SaRouter.match("/home/**").check(check -> {
                try {
                    StpUtil.checkPermission("首页");
                } catch (Exception e) {
                    check.back(handlerException.handlerRolePermission(e));
                }
            });


            /**
             * 账务管理 菜单权限
             */
            SaRouter.match("/financial/**").check(check -> {
                try {
                    StpUtil.checkPermission("财务管理");
                } catch (Exception e) {
                    check.back(handlerException.handlerRolePermission(e));
                }
            });

            /**
             * 管理员管理 菜单权限
             */
            SaRouter.match("/admin/**").check(check -> {
                try {
                    StpUtil.checkPermission("管理员管理");
                } catch (Exception e) {
                    check.back(handlerException.handlerRolePermission(e));
                }
            });

            SaRouter.match("/admin/delAdmin").check(check -> {
                try {
                    StpUtil.checkPermission("删除管理员");
                } catch (Exception e) {
                    check.back(handlerException.handlerRolePermission(e));
                }
            });

            SaRouter.match("/admin/addNewAdmin").check(check -> {
                try {
                    StpUtil.checkPermission("添加管理员");
                } catch (Exception e) {
                    check.back(handlerException.handlerRolePermission(e));
                }
            });

            /**
             * 公告管理 菜单权限
             */
            SaRouter.match("/announcement/**").check(check -> {
                try {
                    StpUtil.checkPermission("公告管理");
                } catch (Exception e) {
                    check.back(handlerException.handlerRolePermission(e));
                }
            });


            /**
             * 维修员管理 菜单权限
             */
            SaRouter.match("/worker/**").check(check -> {
                try {
                    StpUtil.checkPermission("维修员管理");
                } catch (Exception e) {
                    check.back(handlerException.handlerRolePermission(e));
                }
            });


            /**
             * 维修员管理 菜单权限
             */
            SaRouter.match("/user/**").check(check -> {
                try {
                    StpUtil.checkPermission("普通用户管理");
                } catch (Exception e) {
                    check.back(handlerException.handlerRolePermission(e));
                }
            });

            /**
             * 订单管理 菜单权限
             */
            SaRouter.match("/orders/**").check(check -> {
                try {
                    StpUtil.checkPermission("订单管理");
                } catch (Exception e) {
                    check.back(handlerException.handlerRolePermission(e));
                }
            });

            /**
             * 权限管理 菜单权限
             */
            SaRouter.match("/permission/**").notMatch("/permission/pullRole")
                    .check(check -> {
                        try {
                            StpUtil.checkPermission("权限管理");
                        } catch (Exception e) {
                            check.back(handlerException.handlerRolePermission(e));
                        }
                    });

            SaRouter.match("/permission/pullRole").check(check -> {
                try {
                    StpUtil.checkPermissionOr("权限管理", "添加管理员");
                } catch (Exception e) {
                    check.back(handlerException.handlerRolePermission(e));
                }
            });

        });

        // 异常处理函数：每次认证函数发生异常时执行此函数
        saServletFilter.setError(e -> {
            System.out.println("---------- 进入Sa-Token异常处理 -----------");
            return ResponseUtil.setE(ResponseEnum.FAIL).setData(e.getMessage()).toJSONString();
        });

        // 前置函数：在每次认证函数之前执行
        saServletFilter.setBeforeAuth(run -> {
            // 查看跨域请求的地址
            //System.out.println(SaHolder.getRequest().getHeader("Origin"));
            // ---------- 设置跨域响应头 ----------
            SaHolder.getResponse()
                    // 允许指定域访问跨域资源
                    .setHeader("Access-Control-Allow-Origin", "http://localhost:3001")
                    .setHeader("Access-Control-Allow-Origin", "http://127.0.0.1:3001")
                    // 允许所有请求方式
                    .setHeader("Access-Control-Allow-Methods", "POST, GET, OPTIONS, DELETE")
                    // 有效时间
                    .setHeader("Access-Control-Max-Age", "3600")
                    // 允许的header参数
                    .setHeader("Access-Control-Allow-Headers", "*");
        });
        return saServletFilter;
    }
}
